package com.opengate.base.currentuser.resolver;

import com.opengate.base.asserts.AssertHelper;
import com.opengate.base.constants.AppConstants;
import com.opengate.base.currentuser.annotation.CurrentUser;
import com.opengate.base.currentuser.model.CurrentUserDto;
import org.springframework.core.MethodParameter;
import org.springframework.web.bind.support.WebDataBinderFactory;
import org.springframework.web.context.request.NativeWebRequest;
import org.springframework.web.method.support.HandlerMethodArgumentResolver;
import org.springframework.web.method.support.ModelAndViewContainer;

/**
 * @author 曳戈泰尔
 * @version 1.0
 * @description 自定义配套解析器
 *     <p>把当前登录用户信息实例化到CurrentUserDto
 *     <p>将request获得的用户信息返回给@CurrentUser
 * @date 2023/3/13 3:22 PM
 */
public class CurrentUserResolver implements HandlerMethodArgumentResolver {

    /**
     * 用于判断是否支持对某种参数的解析
     *
     * @param parameter 参数
     * @return 是否支持
     */
    @Override
    public boolean supportsParameter(MethodParameter parameter) {
        return parameter.getParameterType().isAssignableFrom(CurrentUserDto.class)
                && parameter.hasParameterAnnotation(CurrentUser.class);
    }

    /**
     * 将请求中的参数值解析为某种对象
     *
     * <p>将User数据传递给@CurrentUser
     *
     * @param parameter 参数
     * @param mavContainer 容器
     * @param webRequest 请求
     * @param binderFactory 工厂
     * @return 用户信息
     * @throws Exception 异常
     */
    @Override
    public Object resolveArgument(
            MethodParameter parameter,
            ModelAndViewContainer mavContainer,
            NativeWebRequest webRequest,
            WebDataBinderFactory binderFactory)
            throws Exception {

        // 返回用户信息到currentUser注解
        String userId = webRequest.getHeader(AppConstants.USER_ID);

        // 判断用户是否为空
        AssertHelper.notEmpty(userId, 499, "登录失败，请重试");

        // 将用户信息封装到CurrentUserDto
        return CurrentUserDto.builder().userId(Integer.valueOf(userId)).build();
    }
}
